from .node_load_owl import OwlModelLoader
from .node_objectness import ObjectnessInference
from .node_bbox_vis import BBoxVisualizer

# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
    "OWL_Load_Model": OwlModelLoader,
    "OWL_Objectness_Inference": ObjectnessInference,
    "OWL_BBox_Visualizer": BBoxVisualizer
}

# A dictionary that contains the friendly/humanly readable titles for the nodes
NODE_DISPLAY_NAME_MAPPINGS = {
    "OWL_Load_Model": "Load OWL Model",
    "OWL_Objectness_Inference": "Objectness",
    "OWL_BBox_Visualizer": "Bounding Box Visualizer"
}
